Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: add to arrow convert #15839

Merged
merged 2 commits into from
Mar 26, 2024
Merged

feat: add to arrow convert #15839

merged 2 commits into from
Mar 26, 2024

Conversation

ZENOTME
Copy link
Contributor

@ZENOTME ZENOTME commented Mar 21, 2024

I hereby agree to the terms of the RisingWave Labs, Inc. Contributor License Agreement.

What's changed and what's your intention?

ToArrow version of #15820

Checklist

  • I have written necessary rustdoc comments
  • I have added necessary unit tests and integration tests
  • I have added test labels as necessary. See details.
  • I have added fuzzing tests or opened an issue to track them. (Optional, recommended for new SQL features Sqlsmith: Sql feature generation #7934).
  • My PR contains breaking changes. (If it deprecates some features, please create a tracking issue to remove them in the future).
  • All checks passed in ./risedev check (or alias, ./risedev c)
  • My PR changes performance-critical code. (Please run macro/micro-benchmarks and show the results.)
  • My PR contains critical fixes that are necessary to be merged into the latest release. (Please check out the details)

Documentation

  • My PR needs documentation updates. (Please use the Release note section below to summarize the impact on users)

Release note

If this PR includes changes that directly affect users or other significant modifications relevant to the community, kindly draft a release note to provide a concise summary of these changes. Please prioritize highlighting the impact these changes will have on users.

@ZENOTME ZENOTME changed the title faet: add to arrow convert feat: add to arrow convert Mar 21, 2024
Copy link
Contributor

@wangrunji0408 wangrunji0408 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

Comment on lines 573 to 575
struct DefaultArrowConvert;
impl ToArrowArrayConvert for DefaultArrowConvert {}
static DEFAULT_ARROW_CONVERT: DefaultArrowConvert = DefaultArrowConvert;
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The static variable is unnecessary since the struct is zero-size and can be instantiated at any time.

Suggested change
struct DefaultArrowConvert;
impl ToArrowArrayConvert for DefaultArrowConvert {}
static DEFAULT_ARROW_CONVERT: DefaultArrowConvert = DefaultArrowConvert;
struct DefaultArrowConvert;
impl ToArrowArrayConvert for DefaultArrowConvert {}

_data_type: &arrow_schema::DataType,
array: &DecimalArray,
) -> Result<arrow_array::ArrayRef, ArrayError> {
Ok(Arc::new(arrow_array::LargeBinaryArray::from(array)))
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please add a comment to indicate this is special.

Decimal values are stored as ASCII text representation in a large binary array.

_data_type: &arrow_schema::DataType,
array: &JsonbArray,
) -> Result<arrow_array::ArrayRef, ArrayError> {
Ok(Arc::new(arrow_array::LargeStringArray::from(array)))
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please add a comment to indicate this is special.

JSON values are stored as text representation in a large string array.

Comment on lines +156 to +164
#[inline]
fn int16_to_arrow(
&self,
_data_type: &arrow_schema::DataType,
array: &I16Array,
) -> Result<arrow_array::ArrayRef, ArrayError> {
Ok(Arc::new(arrow_array::Int16Array::from(array)))
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Any reason why we provide an arrow data type but we don't use it at all in the default implementation?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Because it's not needed in the original default implementation. But it may need for some external system, e.g. in the iceberg, we need to know the Decimal type when we want to convert a decimal array in rw to a decimal array.

Copy link
Contributor

@chenzl25 chenzl25 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@ZENOTME ZENOTME added this pull request to the merge queue Mar 26, 2024
Merged via the queue into main with commit d5ac14f Mar 26, 2024
26 of 27 checks passed
@ZENOTME ZENOTME deleted the zj/convert branch March 26, 2024 07:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants